const config = require('../config')
const BlockSchema = require('./schema/BlockSchema');
const BlockModel = require('./BlockModel');
const BlockHeightSchema = require('./schema/BlockHeightSchema');
const BlockHeightModel = require('./BlockHeightModel');
class Storage {
    constructor (db, logger) {
        this.db = db;
        this.logger = logger;
        this.blockModel = new BlockModel(this.db, config.storage.blockCol, BlockSchema, this.logger);
        this.blockHeightModel = new BlockModel(this.db, config.storage.blockHeightCol, BlockHeightSchema, this.logger);
    }
    getBlockCol(){
        return this.blockModel;
    }
    getBlockHeightCol(){
        return this.blockHeightModel;
    }
}

module.exports = Storage;